#ifndef GETTM_HEADER
#define GETTM_HEADER
#include<string>
#include<fstream>
#include<sstream>
#include<iostream>
#include<algorithm>
#include<iterator>
#include<vector>
#include<map>
#include<cstdlib>
#include<ctime>
#include<cmath>
#include "def.hpp"
#include "topicmodel_generic_model.hpp"
#include "topicmodel_lda_abstract.hpp"
#include "topicmodel_lda_gibbs.cpp"

namespace Puza{
  GenericTopicModel* getTopicModel(){
    if(PuzaDEF::Instance()->currentTopicModelType == TOPICMODEL_LDA_GIBBS_SAMPLING){
      return new LDA_Gibbs();
    }
    return NULL;
  }
};

#endif
